package cn.iocoder.yudao.module.ai.service.dataset;

import java.util.*;
import javax.validation.*;
import cn.iocoder.yudao.module.ai.controller.admin.dataset.vo.*;
import cn.iocoder.yudao.module.ai.controller.admin.datasetversion.vo.DatasetQualityTaskSaveReqVO;
import cn.iocoder.yudao.module.ai.controller.admin.datasetversion.vo.DatasetVersionSaveReqVO;
import cn.iocoder.yudao.module.ai.dal.dataobject.dataset.DatasetDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.module.ai.dal.dataobject.datasetreleasefile.DatasetReleaseFileDO;

/**
 * 数据集管理 Service 接口
 *
 * @author 侯俊青
 */
public interface DatasetService {

    /**
     * 创建数据集管理
     *
     * @param saveReqVO 保存信息
     * @return 编号
     */
    Long saveDataset(@Valid DatasetSaveReqVO saveReqVO);

    /**
     * 创建数据集管理
     *
     * @param saveReqVO 保存信息
     * @return 编号
     */
    Long updateDataset(@Valid DatasetUpdateReqVO saveReqVO);

    /**
     * 创建数据集管理
     *
     * @param saveReqVO 保存信息
     * @return 编号
     */
    Long uploadFile(@Valid DatasetUploadFileReqVO saveReqVO);

//    /**
//     * 创建数据集管理
//     *
//     * @param createReqVO 创建信息
//     * @return 编号
//     */
//    Long createDataset(@Valid DatasetSaveReqVO createReqVO);
//
//    /**
//     * 更新数据集管理
//     *
//     * @param updateReqVO 更新信息
//     */
//    void updateDataset(@Valid DatasetSaveReqVO updateReqVO);

    /**
     * 删除数据集管理
     *
     * @param id 编号
     */
    void deleteDataset(Long id);

    /**
     * 获得数据集管理
     *
     * @param id 编号
     * @return 数据集管理
     */
    DatasetDO getDataset(Long id);

    /**
     * 获得数据集管理分页
     *
     * @param pageReqVO 分页查询
     * @return 数据集管理分页
     */
    PageResult<DatasetDO> getDatasetPage(DatasetPageReqVO pageReqVO);

    /**
     * 获得数据集管理分页
     *
     * @param datasetIdList 分页查询
     * @return 数据集管理分页
     */
    Map<Long, DatasetReleaseFileDO> getFirstFileMap(List<Long> datasetIdList);

    /**
     * 更新质检文件数量和状态
     * @param datasetFileQualityReqVO
     * @return
     */
    void updateFileQuality(DatasetFileQualityReqVO datasetFileQualityReqVO);

    /**
     * 数据集发布版本
     * @param datasetVersionSaveReqVO
     */
    void publish(DatasetVersionSaveReqVO datasetVersionSaveReqVO);

    void createQualityTask(@Valid  DatasetQualityTaskSaveReqVO datasetQualityTaskSaveReqVO);
}